package com.douziit.meilianoa.base.utils;



import com.douziit.meilianoa.base.exception.TipException;

import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 拼音工具类
 * @author wh
 * @date 2018/8/31 10:32
 */
public class PinYinUtil {
    //字母Z使用了两个标签，这里有２７个值
    //i, u, v都不做声母, 跟随前面的字母
    private static char[] chartable = {
        '啊', '芭', '擦', '搭', '蛾', '发', '噶', '哈', '哈',
        '击', '喀', '垃', '妈', '拿', '哦', '啪', '期', '然',
        '撒', '塌', '塌', '塌', '挖', '昔', '压', '匝', '座'
    };
    private static char[] alphatableb = {
        'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I',
        'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z'
    };
    private static char[] alphatables = {
        'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i',
        'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z'
    };
    private static int[] table = new int[27];  //初始化
    static {
        for (int i = 0; i < 27; ++i) {
            table[i] = gbValue(chartable[i]);
        }
    }

    /**
     * 根据首字母进行分组
     * @param list 对象数组
     * @param fieldName 数据所在的属性
     */
    public static Map<String,Object> group(List<?> list, String fieldName){
        Map<String,Object> map = new HashMap<>();
        ArrayList<Object> arraylist=new ArrayList<>();
        String[] alphatableb = {
            "A", "B", "C", "D", "E", "F", "G", "H", "I",
            "J", "K", "L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z","#"
        };
        for(String a:alphatableb){
            for (Object obj : list) {//为了排序都返回大写字母
                try {
                    String firstLetter = fieldName.substring(0,1).toUpperCase();
                    String getter = "get" +firstLetter + fieldName.substring(1);
                    Method method = obj.getClass().getMethod(getter);
                    Object value = method.invoke(obj);
                    if (a.equals(String2AlphaFirst(value.toString(), "b"))) {
                        arraylist.add(obj);
                    }
                } catch (Exception e) {
                    throw new TipException("根据首字母进行分组时出错");
                }

            }
            map.put(a,arraylist);
            arraylist=new ArrayList<>();
        }
        return map;
    }
    //主函数,输入字符,得到他的声母,
    //英文字母返回对应的大小写字母，数字返回#
    //其他非简体汉字返回 '0'  按参数
    private static char Char2Alpha(char ch,String type) {
        if (ch >= '0' && ch <= '9') {
            return '#';
        }
        if (ch >= 'a' && ch <= 'z') {
            return (char) (ch - 'a' + 'A');//为了按字母排序先返回大写字母
        }
        // return ch;
        if (ch >= 'A' && ch <= 'Z') {
            return ch;
        }
        int gb = gbValue(ch);
        if (gb < table[0])
            return '#';
        int i;
        for (i = 0; i < 26; ++i) {
            if (match(i, gb))
                break;
        }

        if (i >= 26){
            return '#';
        } else{
            if("b".equals(type)){//大写
                return alphatableb[i];
            }else{//小写
                return alphatables[i];
            }
        }
    }
    //根据一个包含汉字的字符串返回一个汉字拼音首字母的字符串
    private static String String2Alpha(String SourceStr,String type) {
        StringBuilder Result = new StringBuilder();
        int StrLength = SourceStr.length();
        int i;
        try {
            for (i = 0; i < StrLength; i++) {
                Result.append(Char2Alpha(SourceStr.charAt(i), type));
            }
        } catch (Exception e) {
            Result = new StringBuilder();
        }
        return Result.toString();
    }
    //根据一个包含汉字的字符串返回第一个汉字拼音首字母的字符串
    private static String String2AlphaFirst(String SourceStr,String type) {
        String Result = "";
        try {
            Result += Char2Alpha(SourceStr.charAt(0),type);
        } catch (Exception e) {
            Result = "";
        }
        return Result;
    }
    private static boolean match(int i, int gb) {
        if (gb < table[i])
            return false;
        int j = i + 1;

        //字母Z使用了两个标签
        while (j < 26 && (table[j] == table[i]))
            ++j;
        if (j == 26)
            return gb <= table[j];
        else
            return gb < table[j];
    }

    //取出汉字的编码
    private static int gbValue(char ch) {
        String str = "";
        str += ch;
        try {
            byte[] bytes = str.getBytes("GBK");
            if (bytes.length < 2)
                return 0;
            return (bytes[0] << 8 & 0xff00) + (bytes[1] & 0xff);
        } catch (Exception e) {
            return 0;
        }
    }


}
